ApplyMomentum ================= 对权重张量执行 Momentum/改进动量优化更新。 .. math:: \begin{aligned} accu_t &= moment \cdot accu_{t-1} + g_t \\ update_t &= \begin{cases} (accu_t \cdot moment + g_t), & \text{if nesterov = True} \\ accu_t, & \text{otherwise} \end{cases} \\ weight_t &= weight_{t-1} - learning\_rate \cdot update_t \end{aligned} 输入: - **weight** - 待更新权重张量首地址。 - **accumulate** - 动量累积张量首地址。 - **gradient** - 梯度张量首地址。 - **nesterov** - 是否启用 Nesterov 动量。 - **float_addr** - 浮点数地址,长度2个float数据,第一个数据是学习率(learning_rate),第二个是动量系数(moment)。 - **int_addr** - 整型地址,长度2个int数据,第一个是参与计算的起始索引(start),第二个是参与计算的结束索引(end)。其中start是闭区间,end是开区间。 - **core_mask(int, 可选)** - 核掩码(仅适用于共享存储版本)。 输出: - **weight** - 原地写回更新后的权重张量。 - **accumulate** - 原地写回更新后的动量张量。 支持平台: ``FT78NE`` ``MT7004`` .. note:: - FT78NE 支持 fp32 数据类型。 - MT7004 支持 fp16、fp32 数据类型。 **共享存储版本:** .. c:function:: void hp_apply_momentum_s(half *weight, half *accumulate,const half *gradient, bool nesterov, float *float_addr, int *int_addr, int core_mask) .. c:function:: void fp_apply_momentum_s(float *weight, float *accumulate,const float *gradient, bool nesterov, float *float_addr, int *int_addr, int core_mask) **C调用示例:** .. code-block:: c :linenos: :emphasize-lines: 21 // FT78NE 多核示例 #include #include int main(void) { float *weight = (float *)0xA0000000; // DDR 存储 float *accumulate = (float *)0xB0000000; float *gradient = (float *)0xC0000000; float *float_addr = (float *)0xD0000000; int *int_addr = (int *)0xD10000000; int start = 0; int end = 4096; int_addr[0] = start; int_addr[1] = end; int core_mask = 0xff; float learning_rate = 1e-2f; float moment = 0.99f; float_addr[0] = learning_rate; float_addr[1] = moment; bool nesterov = false; fp_apply_momentum_s(weight, accumulate, gradient, nesterov, float_addr, int_addr, core_mask); return 0; } **私有存储版本:** .. c:function:: void hp_apply_momentum_p(half *weight, half *accumulate,const half *gradient, bool nesterov, float *float_addr, int *int_addr) .. c:function:: void fp_apply_momentum_p(float *weight, float *accumulate,const float *gradient, bool nesterov, float *float_addr, int *int_addr) **C调用示例:** .. code-block:: c :linenos: :emphasize-lines: 21 // MT7004 单核示例 #include #include int main(void) { half *weight = (half *)0x10000000; // L2 存储 half *accumulate = (half *)0x10002000; half *gradient = (half *)0x10004000; float *float_addr = (float *)0x10006000; int *int_addr = (int *)0x10007000; int start = 0; int end = 4096; int_addr[0] = start; int_addr[1] = end; int core_mask = 0xff; float learning_rate = 1e-2f; float moment = 0.99f; float_addr[0] = learning_rate; float_addr[1] = moment; bool nesterov = true; hp_apply_momentum_p(weight, accumulate, gradient, nesterov, float_addr, int_addr); return 0; }